import sys

# 从系统参数中获取命令行参数列表
sys_args = sys.argv


# 遍历参数列表查找匹配的参数
def get_env_flag():
    for arg in sys_args:
        if arg.startswith('env:'):
            # 如果找到了以 '-v' 开头的参数，则进一步判断是否包含 'env:test'
            return arg.split(':')[-1]


def get_env_dic(env_flag):
    if env_flag == 'test':
        # 测试环境
        variables = {
            'baseUrl': 'http://127.0.0.1:6605/mock',
        }

    else:
        # 线上环境 env == 'prod'
        variables = {
            'baseUrl': 'http://xxxxx:6605/mock',
        }

    return variables


env = get_env_flag()

env_vars = get_env_dic(env)
